#!/usr/bin/python3
# -*- coding:utf-8 -*-
""""
   @Author: 叶建伟
   @Email: yejw46507@hundsun.com
   @Time: 2022/6/7 17:07
   @File: 专户产品管理_修改.py
"""
import time

from 线下客户管理.线下客户共用脚本.open_menu import open_menu, arm_form_fill
from 基础公共脚本.步骤检查_hui2 import check_table, check_table_rows,FE
from 基础公共脚本.common_util import auto_fill_form,refresh_click

def special_product_update(input_data=None, ui_expect=None):
    # 进入 风险等级查询 菜单 线下客户管理/线下客户管理
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '线下客户管理/专户产品管理'})
    # 资源设置搜索框表单填入
    FE.table().click_button_in_cell({'产品名称': input_data.get("产品名称")}, "操作", '修改', True)
    time.sleep(1)
    auto_fill_form(text="产品名称", input_data=input_data)
    FE.button("提交").click()
    time.sleep(1)
    refresh_click()
    arm_form_fill({"产品名称": input_data.get('产品名称')})

    # 检查查询结果
    check_table_rows(input_data=None, ui_expect=ui_expect)

def special_product_update_cancel(input_data=None, ui_expect=None):
    # 进入 风险等级查询 菜单 线下客户管理/线下客户管理
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '线下客户管理/专户产品管理'})
    # 资源设置搜索框表单填入
    FE.table().click_button_in_cell({'产品名称': input_data.get("产品名称")}, "操作", '修改', True)
    time.sleep(1)
    auto_fill_form(text="产品名称", input_data=input_data)
    FE.button(idname={'xpath':'//div[@class="btn-footer"]/button/span[text()="取消"]'}).click()
    time.sleep(1)
    refresh_click()
    arm_form_fill({"产品名称": input_data.get('产品名称')})
    # 检查查询结果
    check_table_rows(input_data=None, ui_expect=ui_expect)